package com.mzq.langchain4j_springboot.web.upload;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.PutObjectResult;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.UUID;

@Component    //将bean放入IOC容器中管理
public class AliOSSUtils {

    private final AliOSSProperties aliOSSProperties;

    @Autowired    //通过注入IOC容器中的Bean对象对本类中的aliOSSProperties进行赋值
    public AliOSSUtils(AliOSSProperties aliOSSProperties) {
        this.aliOSSProperties = aliOSSProperties;
    }

    /**
     * 实现上传图片到OSS  第一步
     */
    public String upload(MultipartFile file) throws IOException {
        // 获取上传的文件的输入流
        InputStream inputStream = file.getInputStream();

        // 避免文件覆盖
        String originalFilename = file.getOriginalFilename();
        String fileName = UUID.randomUUID().toString() + originalFilename.substring(originalFilename.lastIndexOf("."));

        // 上传文件到 OSS
        OSS ossClient = new OSSClientBuilder().build(
                aliOSSProperties.getEndpoint(),
                aliOSSProperties.getAccessKeyId(),
                aliOSSProperties.getAccessKeySecret()
        );

        PutObjectResult putObjectResult = ossClient.putObject(aliOSSProperties.getBucketName(), fileName, inputStream);

        // 文件访问路径
        String url = aliOSSProperties.getEndpoint()
                .split("//")[0] + "//" + aliOSSProperties.getBucketName()
                + "." + aliOSSProperties.getEndpoint().split("//")[1] + "/" + fileName;


        // 关闭ossClient
        ossClient.shutdown();


        return url; // 把上传到oss的路径返回
    }
}
